# -*- coding:utf-8 -*-
from sqlite_cli import SQLiteCLI
from colorama import init
import argparse

# 初始化 colorama
init(strip=False, convert=True)

def parse_args():
    parser = argparse.ArgumentParser(description='SQLite CLI 工具')
    parser.add_argument('database', nargs='?', default=':memory:',
                      help='数据库文件路径 (默认使用内存数据库)')
    return parser.parse_args()

if __name__ == "__main__":
    args = parse_args()
    cli = SQLiteCLI(args.database)
    cli.run()